package com.userWallet;

import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;

public class CodeGenerator {
    public static void main(String[] args) {

        // 设置数据库相关配置
        AutoGenerator autoGenerator = new AutoGenerator();


        DataSourceConfig dataSource = new DataSourceConfig();
        dataSource.setDriverName("com.mysql.cj.jdbc.Driver");
        dataSource.setUrl("jdbc:mysql://localhost:3306/user_wallet?serverTimezone=UTC");
        dataSource.setUsername("root");
        dataSource.setPassword("123456");

        autoGenerator.setDataSource(dataSource);


        // 设置全局配置
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setOutputDir(System.getProperty("user.dir") + "/src/main/java");    // 设置代码生成位置
        globalConfig.setOpen(false);            // 设置生成完毕后是否打开生成代码所在的目录
        globalConfig.setAuthor("崔");    // 设置作者
        globalConfig.setFileOverride(true);    // 设置是否覆盖原始生成的文件
        globalConfig.setMapperName("%sMapper");   // 设置数据层接口名， %s 为占位符，指代模块名称
//        globalConfig.setIdType(IdType.ASSIGN_ID);  // 设置 Id 生成策略
        autoGenerator.setGlobalConfig(globalConfig);

        
        // 设置包名相关配置
        PackageConfig packageInfo = new PackageConfig();
        packageInfo.setParent("com.userWallet");   //  设置生成的包名，与代码所在的位置不冲突，二者叠加组成完整路径
        packageInfo.setEntity("domain");    // 设置实体类包名
        packageInfo.setMapper("dao");      // 设置数据层包名
        autoGenerator.setPackageInfo(packageInfo);


        // 策略设置
        StrategyConfig strategyConfig = new StrategyConfig();
//        strategyConfig.setInclude("tbl_user");      // 设置当前参与生成的表名，参数为可变参数
//        strategyConfig.setTablePrefix("tbl_");      // 设置数据库表的前缀名称，模块名 = 数据库表明_前缀名  例如: User = tbl_user
        strategyConfig.setRestControllerStyle(true);   // 设置rest风格
//        strategyConfig.setVersionFieldName("version");    // 设置乐观锁字段名
//        strategyConfig.setLogicDeleteFieldName("deleted");   // 设置逻辑删除字段名
        strategyConfig.setEntityLombokModel(true);     // 设置是否启用 lombok
        autoGenerator.setStrategy(strategyConfig);

        autoGenerator.execute();


    }
}
